Model: "vgg16"
__________________________________________________________________________________________________
Layer (type) Output Shape Param # Connected to
==================================================================================================
input_1 (InputLayer) [(None, 192, 640, 6) 0
__________________________________________________________________________________________________
block_1_conv_1 (Conv2D) (None, 96, 320, 64) 18880 input_1[0][0]
__________________________________________________________________________________________________
batch_normalization (BatchNorma (None, 96, 320, 64) 256 block_1_conv_1[0][0]
__________________________________________________________________________________________________
leaky_re_lu (LeakyReLU) (None, 96, 320, 64) 0 batch_normalization[0][0]
__________________________________________________________________________________________________
block_2_conv_1 (Conv2D) (None, 48, 160, 128) 204928 leaky_re_lu[0][0]
__________________________________________________________________________________________________
batch_normalization_1 (BatchNor (None, 48, 160, 128) 512 block_2_conv_1[0][0]
__________________________________________________________________________________________________
leaky_re_lu_1 (LeakyReLU) (None, 48, 160, 128) 0 batch_normalization_1[0][0]
__________________________________________________________________________________________________
block_3_conv_1 (Conv2D) (None, 24, 80, 256) 819456 leaky_re_lu_1[0][0]
__________________________________________________________________________________________________
batch_normalization_2 (BatchNor (None, 24, 80, 256) 1024 block_3_conv_1[0][0]
__________________________________________________________________________________________________
leaky_re_lu_2 (LeakyReLU) (None, 24, 80, 256) 0 batch_normalization_2[0][0]
__________________________________________________________________________________________________
block_3_conv_2 (Conv2D) (None, 24, 80, 256) 590080 leaky_re_lu_2[0][0]
__________________________________________________________________________________________________
batch_normalization_3 (BatchNor (None, 24, 80, 256) 1024 block_3_conv_2[0][0]
__________________________________________________________________________________________________
leaky_re_lu_3 (LeakyReLU) (None, 24, 80, 256) 0 batch_normalization_3[0][0]
__________________________________________________________________________________________________
block_4_conv_1 (Conv2D) (None, 12, 40, 512) 1180160 leaky_re_lu_3[0][0]
__________________________________________________________________________________________________
batch_normalization_4 (BatchNor (None, 12, 40, 512) 2048 block_4_conv_1[0][0]
__________________________________________________________________________________________________
leaky_re_lu_4 (LeakyReLU) (None, 12, 40, 512) 0 batch_normalization_4[0][0]
__________________________________________________________________________________________________
block_4_conv_2 (Conv2D) (None, 12, 40, 512) 2359808 leaky_re_lu_4[0][0]
__________________________________________________________________________________________________
batch_normalization_5 (BatchNor (None, 12, 40, 512) 2048 block_4_conv_2[0][0]
__________________________________________________________________________________________________
leaky_re_lu_5 (LeakyReLU) (None, 12, 40, 512) 0 batch_normalization_5[0][0]
__________________________________________________________________________________________________
block_5_conv_1 (Conv2D) (None, 6, 20, 512) 2359808 leaky_re_lu_5[0][0]
__________________________________________________________________________________________________
batch_normalization_6 (BatchNor (None, 6, 20, 512) 2048 block_5_conv_1[0][0]
__________________________________________________________________________________________________
leaky_re_lu_6 (LeakyReLU) (None, 6, 20, 512) 0 batch_normalization_6[0][0]
__________________________________________________________________________________________________
block_5_conv_2 (Conv2D) (None, 6, 20, 512) 2359808 leaky_re_lu_6[0][0]
__________________________________________________________________________________________________
batch_normalization_7 (BatchNor (None, 6, 20, 512) 2048 block_5_conv_2[0][0]
__________________________________________________________________________________________________
leaky_re_lu_7 (LeakyReLU) (None, 6, 20, 512) 0 batch_normalization_7[0][0]
__________________________________________________________________________________________________
block_6_conv_1 (Conv2D) (None, 3, 10, 1024) 4719616 leaky_re_lu_7[0][0]
__________________________________________________________________________________________________
batch_normalization_8 (BatchNor (None, 3, 10, 1024) 4096 block_6_conv_1[0][0]
__________________________________________________________________________________________________
leaky_re_lu_8 (LeakyReLU) (None, 3, 10, 1024) 0 batch_normalization_8[0][0]
__________________________________________________________________________________________________
block_6_conv_2 (Conv2D) (None, 3, 10, 1024) 9438208 leaky_re_lu_8[0][0]
__________________________________________________________________________________________________
batch_normalization_9 (BatchNor (None, 3, 10, 1024) 4096 block_6_conv_2[0][0]
__________________________________________________________________________________________________
leaky_re_lu_9 (LeakyReLU) (None, 3, 10, 1024) 0 batch_normalization_9[0][0]
__________________________________________________________________________________________________
conv_trans_0 (Conv2DTranspose) (None, 6, 20, 512) 8389120 leaky_re_lu_9[0][0]
__________________________________________________________________________________________________
conv_flow0 (Conv2D) (None, 3, 10, 2) 18434 leaky_re_lu_9[0][0]
__________________________________________________________________________________________________
leaky_re_lu_10 (LeakyReLU) (None, 6, 20, 512) 0 conv_trans_0[0][0]
__________________________________________________________________________________________________
flow_0 (Conv2DTranspose) (None, 6, 20, 2) 66 conv_flow0[0][0]
__________________________________________________________________________________________________
concatenate (Concatenate) (None, 6, 20, 1026) 0 leaky_re_lu_7[0][0]
leaky_re_lu_10[0][0]
flow_0[0][0]
__________________________________________________________________________________________________
conv_trans_1 (Conv2DTranspose) (None, 12, 40, 256) 4202752 concatenate[0][0]
__________________________________________________________________________________________________
conv_flow1 (Conv2D) (None, 6, 20, 2) 18470 concatenate[0][0]
__________________________________________________________________________________________________
leaky_re_lu_11 (LeakyReLU) (None, 12, 40, 256) 0 conv_trans_1[0][0]
__________________________________________________________________________________________________
flow_1 (Conv2DTranspose) (None, 12, 40, 2) 66 conv_flow1[0][0]
__________________________________________________________________________________________________
concatenate_1 (Concatenate) (None, 12, 40, 770) 0 leaky_re_lu_5[0][0]
leaky_re_lu_11[0][0]
flow_1[0][0]
__________________________________________________________________________________________________
conv_trans_2 (Conv2DTranspose) (None, 24, 80, 128) 1577088 concatenate_1[0][0]
__________________________________________________________________________________________________
conv_flow2 (Conv2D) (None, 12, 40, 2) 13862 concatenate_1[0][0]
__________________________________________________________________________________________________
leaky_re_lu_12 (LeakyReLU) (None, 24, 80, 128) 0 conv_trans_2[0][0]
__________________________________________________________________________________________________
flow_2 (Conv2DTranspose) (None, 24, 80, 2) 66 conv_flow2[0][0]
__________________________________________________________________________________________________
concatenate_2 (Concatenate) (None, 24, 80, 386) 0 leaky_re_lu_3[0][0]
leaky_re_lu_12[0][0]
flow_2[0][0]
__________________________________________________________________________________________________
conv_trans_3 (Conv2DTranspose) (None, 48, 160, 64) 395328 concatenate_2[0][0]
__________________________________________________________________________________________________
conv_flow3 (Conv2D) (None, 24, 80, 2) 6950 concatenate_2[0][0]
__________________________________________________________________________________________________
leaky_re_lu_13 (LeakyReLU) (None, 48, 160, 64) 0 conv_trans_3[0][0]
__________________________________________________________________________________________________
flow_3 (Conv2DTranspose) (None, 48, 160, 2) 66 conv_flow3[0][0]
__________________________________________________________________________________________________
concatenate_3 (Concatenate) (None, 48, 160, 194) 0 leaky_re_lu_1[0][0]
leaky_re_lu_13[0][0]
flow_3[0][0]
__________________________________________________________________________________________________
conv_flow4 (Conv2D) (None, 48, 160, 2) 3494 concatenate_3[0][0]
__________________________________________________________________________________________________
flow_4 (Conv2DTranspose) (None, 96, 320, 2) 66 conv_flow4[0][0]
==================================================================================================
Total params: 38,695,780
Trainable params: 38,686,180
Non-trainable params: 9,600
__________________________________________________________________________________________________
Model: "vgg16"
__________________________________________________________________________________________________
Layer (type) Output Shape Param # Connected to
==================================================================================================
input_1 (InputLayer) [(None, 192, 640, 6) 0
__________________________________________________________________________________________________
block_1_conv_1 (Conv2D) (None, 96, 320, 64) 18880 input_1[0][0]
__________________________________________________________________________________________________
batch_normalization (BatchNorma (None, 96, 320, 64) 256 block_1_conv_1[0][0]
__________________________________________________________________________________________________
leaky_re_lu (LeakyReLU) (None, 96, 320, 64) 0 batch_normalization[0][0]
__________________________________________________________________________________________________
block_2_conv_1 (Conv2D) (None, 48, 160, 128) 204928 leaky_re_lu[0][0]
__________________________________________________________________________________________________
batch_normalization_1 (BatchNor (None, 48, 160, 128) 512 block_2_conv_1[0][0]
__________________________________________________________________________________________________
leaky_re_lu_1 (LeakyReLU) (None, 48, 160, 128) 0 batch_normalization_1[0][0]
__________________________________________________________________________________________________
block_3_conv_1 (Conv2D) (None, 24, 80, 256) 819456 leaky_re_lu_1[0][0]
__________________________________________________________________________________________________
batch_normalization_2 (BatchNor (None, 24, 80, 256) 1024 block_3_conv_1[0][0]
__________________________________________________________________________________________________
leaky_re_lu_2 (LeakyReLU) (None, 24, 80, 256) 0 batch_normalization_2[0][0]
__________________________________________________________________________________________________
block_3_conv_2 (Conv2D) (None, 24, 80, 256) 590080 leaky_re_lu_2[0][0]
__________________________________________________________________________________________________
batch_normalization_3 (BatchNor (None, 24, 80, 256) 1024 block_3_conv_2[0][0]
__________________________________________________________________________________________________
leaky_re_lu_3 (LeakyReLU) (None, 24, 80, 256) 0 batch_normalization_3[0][0]
__________________________________________________________________________________________________
block_4_conv_1 (Conv2D) (None, 12, 40, 512) 1180160 leaky_re_lu_3[0][0]
__________________________________________________________________________________________________
batch_normalization_4 (BatchNor (None, 12, 40, 512) 2048 block_4_conv_1[0][0]
__________________________________________________________________________________________________
leaky_re_lu_4 (LeakyReLU) (None, 12, 40, 512) 0 batch_normalization_4[0][0]
__________________________________________________________________________________________________
block_4_conv_2 (Conv2D) (None, 12, 40, 512) 2359808 leaky_re_lu_4[0][0]
__________________________________________________________________________________________________
batch_normalization_5 (BatchNor (None, 12, 40, 512) 2048 block_4_conv_2[0][0]
__________________________________________________________________________________________________
leaky_re_lu_5 (LeakyReLU) (None, 12, 40, 512) 0 batch_normalization_5[0][0]
__________________________________________________________________________________________________
block_5_conv_1 (Conv2D) (None, 6, 20, 512) 2359808 leaky_re_lu_5[0][0]
__________________________________________________________________________________________________
batch_normalization_6 (BatchNor (None, 6, 20, 512) 2048 block_5_conv_1[0][0]
__________________________________________________________________________________________________
leaky_re_lu_6 (LeakyReLU) (None, 6, 20, 512) 0 batch_normalization_6[0][0]
__________________________________________________________________________________________________
block_5_conv_2 (Conv2D) (None, 6, 20, 512) 2359808 leaky_re_lu_6[0][0]
__________________________________________________________________________________________________
batch_normalization_7 (BatchNor (None, 6, 20, 512) 2048 block_5_conv_2[0][0]
__________________________________________________________________________________________________
leaky_re_lu_7 (LeakyReLU) (None, 6, 20, 512) 0 batch_normalization_7[0][0]
__________________________________________________________________________________________________
block_6_conv_1 (Conv2D) (None, 3, 10, 1024) 4719616 leaky_re_lu_7[0][0]
__________________________________________________________________________________________________
batch_normalization_8 (BatchNor (None, 3, 10, 1024) 4096 block_6_conv_1[0][0]
__________________________________________________________________________________________________
leaky_re_lu_8 (LeakyReLU) (None, 3, 10, 1024) 0 batch_normalization_8[0][0]
__________________________________________________________________________________________________
block_6_conv_2 (Conv2D) (None, 3, 10, 1024) 9438208 leaky_re_lu_8[0][0]
__________________________________________________________________________________________________
batch_normalization_9 (BatchNor (None, 3, 10, 1024) 4096 block_6_conv_2[0][0]
__________________________________________________________________________________________________
leaky_re_lu_9 (LeakyReLU) (None, 3, 10, 1024) 0 batch_normalization_9[0][0]
__________________________________________________________________________________________________
conv_trans_0 (Conv2DTranspose) (None, 6, 20, 512) 8389120 leaky_re_lu_9[0][0]
__________________________________________________________________________________________________
conv_flow0 (Conv2D) (None, 3, 10, 2) 18434 leaky_re_lu_9[0][0]
__________________________________________________________________________________________________
leaky_re_lu_10 (LeakyReLU) (None, 6, 20, 512) 0 conv_trans_0[0][0]
__________________________________________________________________________________________________
flow_0 (Conv2DTranspose) (None, 6, 20, 2) 66 conv_flow0[0][0]
__________________________________________________________________________________________________
concatenate (Concatenate) (None, 6, 20, 1026) 0 leaky_re_lu_7[0][0]
leaky_re_lu_10[0][0]
flow_0[0][0]
__________________________________________________________________________________________________
conv_trans_1 (Conv2DTranspose) (None, 12, 40, 256) 4202752 concatenate[0][0]
__________________________________________________________________________________________________
conv_flow1 (Conv2D) (None, 6, 20, 2) 18470 concatenate[0][0]
__________________________________________________________________________________________________
leaky_re_lu_11 (LeakyReLU) (None, 12, 40, 256) 0 conv_trans_1[0][0]
__________________________________________________________________________________________________
flow_1 (Conv2DTranspose) (None, 12, 40, 2) 66 conv_flow1[0][0]
__________________________________________________________________________________________________
concatenate_1 (Concatenate) (None, 12, 40, 770) 0 leaky_re_lu_5[0][0]
leaky_re_lu_11[0][0]
flow_1[0][0]
__________________________________________________________________________________________________
conv_trans_2 (Conv2DTranspose) (None, 24, 80, 128) 1577088 concatenate_1[0][0]
__________________________________________________________________________________________________
conv_flow2 (Conv2D) (None, 12, 40, 2) 13862 concatenate_1[0][0]
__________________________________________________________________________________________________
leaky_re_lu_12 (LeakyReLU) (None, 24, 80, 128) 0 conv_trans_2[0][0]
__________________________________________________________________________________________________
flow_2 (Conv2DTranspose) (None, 24, 80, 2) 66 conv_flow2[0][0]
__________________________________________________________________________________________________
concatenate_2 (Concatenate) (None, 24, 80, 386) 0 leaky_re_lu_3[0][0]
leaky_re_lu_12[0][0]
flow_2[0][0]
__________________________________________________________________________________________________
conv_trans_3 (Conv2DTranspose) (None, 48, 160, 64) 395328 concatenate_2[0][0]
__________________________________________________________________________________________________
conv_flow3 (Conv2D) (None, 24, 80, 2) 6950 concatenate_2[0][0]
__________________________________________________________________________________________________
leaky_re_lu_13 (LeakyReLU) (None, 48, 160, 64) 0 conv_trans_3[0][0]
__________________________________________________________________________________________________
flow_3 (Conv2DTranspose) (None, 48, 160, 2) 66 conv_flow3[0][0]
__________________________________________________________________________________________________
concatenate_3 (Concatenate) (None, 48, 160, 194) 0 leaky_re_lu_1[0][0]
leaky_re_lu_13[0][0]
flow_3[0][0]
__________________________________________________________________________________________________
conv_flow4 (Conv2D) (None, 48, 160, 2) 3494 concatenate_3[0][0]
__________________________________________________________________________________________________
flow_4 (Conv2DTranspose) (None, 96, 320, 2) 66 conv_flow4[0][0]
==================================================================================================
Total params: 38,695,780
Trainable params: 38,686,180
Non-trainable params: 9,600
__________________________________________________________________________________________________